package com.ww.algorithm.question.leetcode;

/**
 * 35.搜索插入位置
 * 给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。
 *
 * 你可以假设数组中无重复元素。
 *
 * 示例 1:
 *
 *     输入: [1,3,5,6], 5
 *     输出: 2
 *
 * 示例 2:
 *
 *     输入: [1,3,5,6], 2
 *     输出: 1
 *
 * 示例 3:
 *
 *     输入: [1,3,5,6], 7
 *     输出: 4
 *
 * 示例 4:
 *
 *     输入: [1,3,5,6], 0
 *     输出: 0
 *
 *
 */
public class Question35 {

    /**
     * 左闭右闭
     * @param nums
     * @param target
     * @return
     */
    public int searchInsert(int[] nums, int target) {
        int n = nums.length;

        int low = 0;
        int high = n - 1;

        while (low <= high) {
            int mid = low + (high - low) / 2;
            if (nums[mid] > target) {
                high = mid - 1;
            } else if (nums[mid] < target) {
                low = mid + 1;
            } else {
                return mid;
            }
        }

        // 数组中未找到
        return high + 1;
    }

    /**
     * 左闭右开
     * @param nums
     * @param target
     * @return
     */
    public int searchInsert2(int[] nums, int target) {
        int left = 0;
        int right = nums.length;
        while (left < right) {
            int mid = left + ((right - left) >> 1);
            if (nums[mid] > target) {
                right = mid; // target在左区间，在[left, mid)中
            } else if (nums[mid] < target) {
                left = mid + 1; // target在右区间，在[mid+1, right)中
            } else {
                return mid;
            }
        }
        return right;
    }
}
